from __future__ import absolute_import

from ESClient.dataframe import DataFrame, Column
from ESClient.client import RestClient
from ESClient.queries import Select, Agg
from ESClient.types import Row
from elasticsearch import Elasticsearch
from elasticsearch import helpers

col = Column

DataframeClient = DataFrame.from_es


class OfficialClient(Elasticsearch):
    def helper_bulk(self, actions):
        helpers.bulk(self, actions=actions)
